#python
#-*- coding: utf-8 -*-

import openai

def ask(prompt="你好，什么是大模型？", 
        base_url="http://localhost:11434/v1",
        api_key="ollama",
        model="deepseek-r1:1.5b",
        temperature=0.7,             # 控制生成多样性
        max_tokens=512               # 最大生成 token 数
        ):
    client = openai.OpenAI(base_url=base_url, api_key=api_key)
    response = client.chat.completions.create(
            model=model,             # 指定模型
            messages=[
                {"role": "system", "content": "你是一个有帮助的助手。"},
                {"role": "user", "content": prompt}
                ],
            temperature=temperature, # 控制生成多样性
            max_tokens=max_tokens    # 最大生成 token 数
            )
    return response.choices[0].message.content


